from collections import OrderedDict

from rest_framework.pagination import PageNumberPagination
from rest_framework.response import Response


# 重写JWT返回结果方法
def jwt_response_payload_handler(token, user=None, request=None):
    """
    自定义jwt认证成功返回数据
    """
    return {
        'token': token,
        'id': user.id,
        'username': user.username
    }

# 自定义分页器
class PageNum(PageNumberPagination):

    page_size_query_param = 'pagesize'
    max_page_size = 10

    def get_paginated_response(self, data):
        # return Response(OrderedDict([
        #     ('count', self.page.paginator.count),
        #     ('list', data),
        #     ('page', self.page.number),
        #     ('pages', self.page.paginator.num_pages),
        #     ('pagesize', self.max_page_size),
        #
        # ]))

        return Response({

            'count': self.page.paginator.count,
            'lists': data,
            'page': self.page.number,
            'pages': self.page.paginator.num_pages,
            'pagesize': self.max_page_size,

        })